'''
Practice PT #2
Example #1
Add if statement to cute_beeps() function.
'''
from botcore import *
from time import sleep
from random import randrange

delay = 0.1
n_led = 0
n_guests = 0
F4 = 349
C5 = 523

beeps = [600, 800, 1000, 1200, 1400]
fanfare = [(F4, 0.4), (25, 0.2), 
            (F4, 0.1), (F4, 0.1), (C5, 0.8)]

def play_song(song):
    for music in song:
        freq = music[0]
        duration = music[1]
        note(freq, duration)

def cute_beeps(move_type):
    if move_type == 'spin':
        for count in range(22):
            f = randrange(100, 1000)
            spkr.pitch(f)
            sleep(0.1)
    else:
        for note in beeps:
            spkr.pitch(note)
            sleep(0.1)
    spkr.off()

def sweep_leds(n_led):
    leds.user_num(n_led, True)
    sleep(delay)
    leds.user_num(n_led, False)

def count_guests(n_guests):
    freq = beeps[n_guests]
    spkr.pitch(freq)
    sleep(0.1)
    spkr.off()
    buttons.was_pressed(0)
    leds.ls_num(n_guests, True)

def go_straight(speed, delay):
    cute_beeps('forward')
    motors.run(LEFT, speed)
    motors.run(RIGHT, speed)
    sleep(delay)

def spin(speed):
    motors.run(LEFT, speed)
    motors.run(RIGHT, -speed)
    cute_beeps('spin')

def note(freq, duration):
    spkr.pitch(freq)
    sleep(duration)
    spkr.off()
    sleep(0.05)

# -- Main Program --
while True:
    sweep_leds(n_led)
    n_led = n_led + 1 
    if n_led == 8:
        n_led = 0

    if buttons.was_pressed(0):
        count_guests(n_guests)
        n_guests = n_guests + 1
        if n_guests == 5:
            break

# Robot walks out and spins
motors.enable(True)
go_straight(60, 3.0)
spin(30)
motors.enable(False)

# Play fanfare song
play_song(fanfare)